# !/usr/bin/env python
# -*- coding:utf-8 -*-
# @FileName  :login_utils.py
# @Time      :2022/5/30 20:38
# @Author    :张志坤
from flask import g
from functools import wraps


# 判断用户是否登录


def login_required(func):
    """
    强制登陆
    就是判断有没有登录  没有登陆的视图需要先进行登录
    """

    @wraps(func)
    def wrapper(*args, **kwargs):
        # 在视图函数执行前判断是否登录
        print('uid--->', g.user_id)
        if g.user_id:
            return func(*args, **kwargs)
        else:
            return {'msg': "请先登录"}, 401
        # if not g.user_id:
        #     return {'message': '用户未授权.'}, 401
        # elif g.is_refresh:
        #     return {'message': 'Do not use refresh token.'}, 403
        # else:
        #     return func(*args, **kwargs)

    return wrapper
